SAE on toy model no resampling-tied weights